import fitz  # PyMuPDF

# 打开PDF文件
doc = fitz.open(f'HCIE-openEuler V1.0 实验手册.pdf')

# # 检查文档是否加密
# if doc.needsPass:
#     print("PDF文件是加密的，需要密码才能打开。")
#     # 这里需要输入密码来解锁文档
#     doc.authenticate("your_password")
# else:
#     print("PDF文件没有加密。")

# 获取当前的权限设置
permissions = doc.permissions
# print(permissions)
# print(fitz.PDF_PERM_COPY)

# 检查是否允许复制
if not (permissions & fitz.PDF_PERM_COPY):
    print("PDF不允许复制。将修改权限以允许复制。")

    # 设置新的权限，允许打印、复制和注释
    new_permissions = fitz.PDF_PERM_ACCESSIBILITY | fitz.PDF_PERM_PRINT | fitz.PDF_PERM_COPY | fitz.PDF_PERM_ANNOTATE

    # 保存修改后的PDF文件
    doc.save("HCIE-openEuler V1.0 实验手册(已修改为允许复制).pdf", permissions=new_permissions)
    print("权限已修改，现在允许复制。")
else:
    print("PDF允许复制。")

# 关闭PDF文件
doc.close()